package com.zwh.spring.demo;

import java.util.Arrays;

import javax.servlet.http.HttpServletRequest;

import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

@Aspect
@Order(10)
@Component
public class WebLogAspect {

	private Logger logger = LoggerFactory.getLogger(getClass());
	
	//记录当前线程处理的开始时间，用于当前业务操作的耗时 
	private ThreadLocal<Long> startTime = new ThreadLocal<>();
	
	@Pointcut("execution(public * com.zwh.spring.demo.web..*.*(..)) || execution(public * com.zwh.spring.demo.api..*.*(..))")
	public void webLog() {}
	
	@Before("webLog()")
	public void doBefore(JoinPoint joinPoint) throws Throwable{
		startTime.set(System.currentTimeMillis());
		
		ServletRequestAttributes attributes = (ServletRequestAttributes)RequestContextHolder.getRequestAttributes();
		HttpServletRequest request = attributes.getRequest();
		
		//记录下请求内容
		logger.info("URL : " + request.getRequestURL().toString());
		logger.info("HTTP_METHOD : " + request.getMethod());
		logger.info("IP : " + request.getRemoteAddr());
		logger.info("CLASS_METHOD : " + joinPoint.getSignature().getDeclaringTypeName() + "." + joinPoint.getSignature().getName());
		logger.info("ARGS : " + Arrays.toString(joinPoint.getArgs()));
		
	}
	
	@AfterReturning(returning="ret", pointcut="webLog()")
	public void doAfterReturning(Object ret) throws Throwable{
		//处理完请求，返回内容
		logger.info("RESPONSE : " + ret);
		try {
			logger.info("SOEND TIME : " + (System.currentTimeMillis() - startTime.get()) + "ms.");			
		}catch(Exception e) {
			e.printStackTrace();
		}
	}
}
